package com.bbcare.comm.ExcelUtil;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Set;

import org.apache.poi.hssf.usermodel.HSSFCell;
import org.apache.poi.hssf.usermodel.HSSFRow;
import org.apache.poi.hssf.usermodel.HSSFSheet;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.CellStyle;
import org.apache.poi.ss.usermodel.Font;
import org.apache.poi.ss.usermodel.IndexedColors;
import org.apache.poi.ss.util.CellRangeAddress;
/**
 * Excel工具类
 * @ClassName: ExcelUtil 
 * @author CXJ
 * @date 2016年8月3日
 */
public class ExcelUtil {
    private static HSSFWorkbook wb;

    private static CellStyle titleStyle; // 标题行样式
    private static Font titleFont; // 标题行字体
    private static CellStyle dateStyle; // 日期行样式
    private static Font dateFont; // 日期行字体
    private static CellStyle headStyle; // 表头行样式
    private static Font headFont; // 表头行字体
    private static CellStyle contentStyle; // 内容行样式
    private static Font contentFont; // 内容行字体

    /**
     * 导出文件
     * 
     * @param setInfo
     * @param outputExcelFileName
     * @return
     * @throws IOException
     */
    public static boolean export2File(ExcelExportData setInfo,
            String outputExcelFileName) throws Exception {
        return FileUtil.write(outputExcelFileName, export2ByteArray(setInfo),true, true);
    }

    /**
     * 导出到byte数组
     * 
     * @param setInfo
     * @return
     * @throws Exception
     */
    public static byte[] export2ByteArray(ExcelExportData setInfo)
            throws Exception {
        return export2Stream(setInfo).toByteArray();
    }

    /**
     * 导出到流
     * 
     * @param setInfo
     * @return
     * @throws Exception
     */
    public static ByteArrayOutputStream export2Stream(ExcelExportData setInfo)
            throws Exception {
        ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
        export2Stream(setInfo, outputStream);
        return outputStream;
    }
   /**
    * 处理导出excel超出一定数量失败问题。
    * @param setInfo
    * @param os
    * @throws Exception
    */
    public static void export2Stream(ExcelExportData setInfo, OutputStream os)
            throws Exception {
        init();
        Set<Entry<String, List<?>>> set = setInfo.getDataMap().entrySet();
        String[] sheetNames = new String[setInfo.getDataMap().size()];
        int sheetNameNum = 0;
        for (Entry<String, List<?>> entry : set) {
            sheetNames[sheetNameNum] = entry.getKey();
            sheetNameNum++;
        }
        HSSFSheet[] sheets = getSheets(setInfo.getDataMap().size(), sheetNames);
        int sheetNum = 0;
        for (Entry<String, List<?>> entry : set) {
            // Sheet
            List<?> objs = entry.getValue();
            // 标题行
            createTableTitleRow(setInfo, sheets, sheetNum);
            // 日期行  (导出日期)
            createTableDateRow(setInfo, sheets, sheetNum);
            // 表头
            if(null != setInfo.getMoreColumns()){
            	creatTableTwoHeadRow(setInfo, sheets, sheetNum);
            }else{
            	creatTableHeadRow(setInfo, sheets, sheetNum);
            }
            // 表体
            creatTableBody(objs,setInfo,sheets,sheetNum);
            
            sheetNum++;
        }
        wb.write(os);
    }
    
    /**
     * @Description: 初始化
     */
    private static void init() {
        wb = new HSSFWorkbook();

        titleFont = wb.createFont();
        titleStyle = wb.createCellStyle();
        dateStyle = wb.createCellStyle();
        dateFont = wb.createFont();
        headStyle = wb.createCellStyle();
        headFont = wb.createFont();
        contentStyle = wb.createCellStyle();
        contentFont = wb.createFont();

        initTitleCellStyle();
        initTitleFont();
        initDateCellStyle();
        initDateFont();
        initHeadCellStyle();
        initHeadFont();
        initContentCellStyle();
        initContentFont();
    }

    /**
     * @Description: 自动调整列宽
     */
    private static void adjustColumnSize(HSSFSheet[] sheets, int sheetNum,
            String[] fieldNames) {
        for (int i = 0; i < fieldNames.length + 1; i++) {
            sheets[sheetNum].autoSizeColumn(i, true);
        }
    }

    /**
     * @Description: 创建标题行(需合并单元格)
     */
    private static void createTableTitleRow(ExcelExportData setInfo,
            HSSFSheet[] sheets, int sheetNum) {
        CellRangeAddress titleRange = new CellRangeAddress(0, 0, 0, setInfo
                .getFieldNames().get(sheetNum).length);
        sheets[sheetNum].addMergedRegion(titleRange);
        HSSFRow titleRow = sheets[sheetNum].createRow(0);
        titleRow.setHeight((short) 800);
        HSSFCell titleCell = titleRow.createCell(0);
        titleCell.setCellStyle(titleStyle);
        titleCell.setCellValue(setInfo.getTitles()[sheetNum]);
    }

    /**
     * @Description: 创建日期行(需合并单元格)
     */
	private static void createTableDateRow(ExcelExportData setInfo,
            HSSFSheet[] sheets, int sheetNum) {
        CellRangeAddress dateRange = new CellRangeAddress(1, 1, 0, setInfo
                .getFieldNames().get(sheetNum).length);
        sheets[sheetNum].addMergedRegion(dateRange);
        HSSFRow dateRow = sheets[sheetNum].createRow(1);
        dateRow.setHeight((short) 350);
        HSSFCell dateCell = dateRow.createCell(0);
        dateCell.setCellStyle(dateStyle);
        if (null==setInfo.getExportExcelDate()||"".equals(setInfo.getExportExcelDate())) {
        	dateCell.setCellValue("日期："+new SimpleDateFormat("yyyy-MM-dd").format(new Date()));
		}else{
			dateCell.setCellValue("日期："+setInfo.getExportExcelDate());
		}
    }

    /**
     * @Description: 创建表头行(需合并单元格)
     */
    private static void creatTableHeadRow(ExcelExportData setInfo,
            HSSFSheet[] sheets, int sheetNum) {
        // 表头
        HSSFRow headRow = sheets[sheetNum].createRow(2);
        headRow.setHeight((short) 350);
        // 序号列
        HSSFCell snCell = headRow.createCell(0);
        snCell.setCellStyle(headStyle);
        snCell.setCellValue("序号");
        // 列头名称
        for (int num = 1, len = setInfo.getColumnNames().get(sheetNum).length; num <= len; num++) {
            HSSFCell headCell = headRow.createCell(num);
            headCell.setCellStyle(headStyle);
            headCell.setCellValue(setInfo.getColumnNames().get(sheetNum)[num - 1]);
        }
    }
    
    /**
     * @Description: 创建表头行(只针对两行表头)(需合并单元格)
     */
    private static void creatTableTwoHeadRow(ExcelExportData setInfo,
            HSSFSheet[] sheets, int sheetNum) {

		HSSFRow headRowOne = sheets[sheetNum].createRow(2);
		headRowOne.setHeight((short) 350);
		 
		HSSFRow headRowTow = sheets[sheetNum].createRow(3);
		headRowTow.setHeight((short) 350);
        //获取表头信息
		Map<String, String[]> map = new LinkedHashMap<String, String[]>();
		map = setInfo.getMoreColumns();
		//列号从0开始
		int col = 0;
		
		HSSFCell snCell;
		for(String key : map.keySet()){
			if(map.get(key).length > 1){
				//复合表头
				sheets[sheetNum].addMergedRegion(new CellRangeAddress(2, 2, col, col-1+map.get(key).length));
		    	//设置合并单元格的格式
		    	for(int i = col ; i < col+map.get(key).length;i++){
		    		//先设置第一行表头值
		    		snCell = headRowOne.createCell(i);
		    		snCell.setCellStyle(headStyle);
		    		snCell.setCellValue(key);
		    		//设置合并单元格的子节点的值和对象
		    		snCell = headRowTow.createCell(i);
		        	snCell.setCellStyle(headStyle);
		        	snCell.setCellValue(map.get(key)[i-col]);
		    	}
		    	col += map.get(key).length;
			}else{
				//单个表头  合并单元格
				CellRangeAddress dateRange = new CellRangeAddress(2, 3, col, col);
				sheets[sheetNum].addMergedRegion(dateRange);
				
		        snCell = headRowOne.createCell(col);
		        snCell.setCellStyle(headStyle);
		        snCell.setCellValue(key);
		        
		        snCell = headRowTow.createCell(col);
		        snCell.setCellStyle(headStyle);
		        col+=1;
			}
		
		}
    }


    private static void creatTableBody(List<?> objs,ExcelExportData setInfo,
            HSSFSheet[] sheets, int sheetNum){
    	 String[] fieldNames = setInfo.getFieldNames().get(sheetNum);
    	 //如果两行表头从第四行开始（0开始计数）写数据否则第三行
         int rowNum = null == setInfo.getMoreColumns() ? 3 : 4;
         for (Object obj : objs) {
             HSSFRow contentRow = sheets[sheetNum].createRow(rowNum);
             contentRow.setHeight((short) 300);
             HSSFCell[] cells = getCells(contentRow, setInfo.getFieldNames().get(sheetNum).length ,setInfo);
             int cellNum = 1; // 去掉一列序号，因此从1开始
             if (fieldNames != null) {
                 for (int num = 0; num < fieldNames.length; num++) {
                     Object value = ReflectionUtil.invokeGetterMethod(obj,fieldNames[num]);
                     cells[cellNum].setCellValue(value == null ? "" : value.toString());
                      cellNum++;
                 }
             }
             rowNum++;
         }
         adjustColumnSize(sheets, sheetNum, fieldNames); // 自动调整列宽
    }
    /**
     * @Description: 创建所有的Sheet
     */
    private static HSSFSheet[] getSheets(int num, String[] names) {
        HSSFSheet[] sheets = new HSSFSheet[num];
        for (int i = 0; i < num; i++) {
            sheets[i] = wb.createSheet(names[i]);
        }
        return sheets;
    }

    /**
     * @Description: 创建内容行的每一列(附加一列序号)
     */
    private static HSSFCell[] getCells(HSSFRow contentRow, int num,ExcelExportData setInfo) {
        HSSFCell[] cells = new HSSFCell[num + 1];
        for (int i = 0, len = cells.length; i < len; i++) {
            cells[i] = contentRow.createCell(i);
            cells[i].setCellStyle(contentStyle);
        }
        // 设置序号列值，因为出去标题行 ，日期行 所以-2,如果2行表头还要再减1
        int subtractRow = null == setInfo.getMoreColumns() ? 2 : 3;
        cells[0].setCellValue(contentRow.getRowNum() - subtractRow);
        return cells;
    }

    /**
     * @Description: 初始化标题行样式
     */
    private static void initTitleCellStyle() {
        titleStyle.setAlignment(CellStyle.ALIGN_CENTER);
        titleStyle.setVerticalAlignment(CellStyle.VERTICAL_CENTER);
        titleStyle.setFont(titleFont);
        titleStyle.setFillBackgroundColor(IndexedColors.SKY_BLUE.index);
    }

    /**
     * @Description: 初始化日期行样式
     */
    private static void initDateCellStyle() {
        dateStyle.setAlignment(CellStyle.ALIGN_CENTER_SELECTION);
        dateStyle.setVerticalAlignment(CellStyle.VERTICAL_CENTER);
        //dateStyle.setFont(dateFont);
        dateStyle.setFont(headFont);
        dateStyle.setFillBackgroundColor(IndexedColors.SKY_BLUE.index);
    }

    /**
     * @Description: 初始化表头行样式
     */
    private static void initHeadCellStyle() {
        headStyle.setAlignment(CellStyle.ALIGN_CENTER);
        headStyle.setVerticalAlignment(CellStyle.VERTICAL_CENTER);
        headStyle.setFont(headFont);
        headStyle.setFillBackgroundColor(IndexedColors.YELLOW.index);
        headStyle.setBorderTop(CellStyle.BORDER_MEDIUM);
        headStyle.setBorderBottom(CellStyle.BORDER_THIN);
        headStyle.setBorderLeft(CellStyle.BORDER_THIN);
        headStyle.setBorderRight(CellStyle.BORDER_THIN);
        headStyle.setTopBorderColor(IndexedColors.BLUE.index);
        headStyle.setBottomBorderColor(IndexedColors.BLUE.index);
        headStyle.setLeftBorderColor(IndexedColors.BLUE.index);
        headStyle.setRightBorderColor(IndexedColors.BLUE.index);
    }

    /**
     * @Description: 初始化内容行样式
     */
    private static void initContentCellStyle() {
        contentStyle.setAlignment(CellStyle.ALIGN_CENTER);
        contentStyle.setVerticalAlignment(CellStyle.VERTICAL_CENTER);
        contentStyle.setFont(contentFont);
        contentStyle.setBorderTop(CellStyle.BORDER_THIN);
        contentStyle.setBorderBottom(CellStyle.BORDER_THIN);
        contentStyle.setBorderLeft(CellStyle.BORDER_THIN);
        contentStyle.setBorderRight(CellStyle.BORDER_THIN);
        contentStyle.setTopBorderColor(IndexedColors.BLUE.index);
        contentStyle.setBottomBorderColor(IndexedColors.BLUE.index);
        contentStyle.setLeftBorderColor(IndexedColors.BLUE.index);
        contentStyle.setRightBorderColor(IndexedColors.BLUE.index);
        contentStyle.setWrapText(true); // 字段换行
    }

    /**
     * @Description: 初始化标题行字体
     */
    private static void initTitleFont() {
        titleFont.setFontName("华文楷体");
        titleFont.setFontHeightInPoints((short) 20);
        titleFont.setBoldweight(Font.BOLDWEIGHT_BOLD);
        titleFont.setCharSet(Font.DEFAULT_CHARSET);
        titleFont.setColor(IndexedColors.BLUE_GREY.index);
    }

    /**
     * @Description: 初始化日期行字体
     */
    private static void initDateFont() {
        dateFont.setFontName("隶书");
        dateFont.setFontHeightInPoints((short) 10);
        dateFont.setBoldweight(Font.BOLDWEIGHT_BOLD);
        dateFont.setCharSet(Font.DEFAULT_CHARSET);
        dateFont.setColor(IndexedColors.BLUE_GREY.index);
    }

    /**
     * @Description: 初始化表头行字体
     */
    private static void initHeadFont() {
        headFont.setFontName("宋体");
        headFont.setFontHeightInPoints((short) 10);
        headFont.setBoldweight(Font.BOLDWEIGHT_BOLD);
        headFont.setCharSet(Font.DEFAULT_CHARSET);
        headFont.setColor(IndexedColors.BLUE_GREY.index);
    }

    /**
     * @Description: 初始化内容行字体
     */
    private static void initContentFont() {
        contentFont.setFontName("宋体");
        contentFont.setFontHeightInPoints((short) 10);
        contentFont.setBoldweight(Font.BOLDWEIGHT_NORMAL);
        contentFont.setCharSet(Font.DEFAULT_CHARSET);
        contentFont.setColor(IndexedColors.BLUE_GREY.index);
    }

    /**
     * Excel导出数据类
     *
     */
    public static class ExcelExportData {
        /**
         * 导出数据 key:String 表示每个Sheet的名称 <br/>
         * value:List<?> 表示每个Sheet里的所有数据行<br/>
         * 注：List集合中存放的是对象列表,最好使用LinkList
         */
        private LinkedHashMap<String, List<?>> dataMap;

        /**
         * 每个Sheet里的顶部大标题
         */
        private String[] titles;

        /**
         * 单个sheet里的数据列标题
         */
        private List<String[]> columnNames;

        /**
         * 单个sheet里每行数据的列对应的对象属性名称
         */
        private List<String[]> fieldNames;
        
        private String exportExcelDate;//导出日期
        
        /**
         * 单个sheet里的多行数据列标题
         */
        private Map<String , String[]> moreColumns;
        
        
        

        public String getExportExcelDate() {
			return exportExcelDate;
		}

		public void setExportExcelDate(String exportExcelDate) {
			this.exportExcelDate = exportExcelDate;
		}

		public List<String[]> getFieldNames() {
            return fieldNames;
        }

        public void setFieldNames(List<String[]> fieldNames) {
            this.fieldNames = fieldNames;
        }

        public String[] getTitles() {
            return titles;
        }

        public void setTitles(String[] titles) {
            this.titles = titles;
        }

        public List<String[]> getColumnNames() {
            return columnNames;
        }

        public void setColumnNames(List<String[]> columnNames) {
            this.columnNames = columnNames;
        }

        public LinkedHashMap<String, List<?>> getDataMap() {
            return dataMap;
        }

        public void setDataMap(LinkedHashMap<String, List<?>> dataMap) {
            this.dataMap = dataMap;
        }

		public Map<String, String[]> getMoreColumns() {
			return moreColumns;
		}

		public void setMoreColumns(Map<String, String[]> moreColumns) {
			this.moreColumns = moreColumns;
		}

	
		

    }
}